#pragma once

#include <iostream>
#include <cerrno>
#include <cstring>
#include <assert.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/shm.h>
#include <sys/ipc.h>

using namespace std;

#define PATHNAME "."
#define PROJID 0x666

#define SIZE 1024

int getKey()
{
    int key = ftok(PATHNAME, PROJID);
    if(key == -1)
    {
        cerr << errno << ": " << strerror(errno) << endl;
        exit(-1);
    }
    return key;
}

string toHex(int key)
{
    char buffer[64];
    snprintf(buffer, sizeof(buffer), "0x%x", key);
    
    return buffer;
}

//  IPC_CREAT and IPC_EXCL
// 单独使用IPC_CREAT: 创建一个共享内存，如果共享内存不存在，就创建之，如果已经存在，获取已经存在的共享内存并返回
// IPC_EXCL不能单独使用，一般都要配合IPC_CREAT
// IPC_CREAT | IPC_EXCL: 创建一个共享内存，如果共享内存不存在，就创建之, 如果已经存在，则立马出错返回 -- 如果创建成功，对应的shm，一定是最新的！

int creatShmHelper(key_t key, int size, int flag)
{
    int shmid = shmget(key, size, flag);
    if(shmid == -1)
    {
        cerr << errno << ": " << strerror(errno) << endl;
        exit(-1);
    }
    return shmid;
}

int creatShm(key_t key, int size)
{
    umask(0);
    return creatShmHelper(key, size, IPC_CREAT | IPC_EXCL | 0666);
}

int getShm(key_t key, int size)
{
    return creatShmHelper(key, size, IPC_CREAT);
}
char* attachShm(int shmid)
{
    //返回的是，共享内存的起始地址
    char *start = (char*)shmat(shmid, nullptr, 0);//第三个参数为0，默认就是读写权限
    return start;
}
void detachShm(char* str)
{
    int n = shmdt(str);
    assert(n != -1);
    (void)n;
}

void delShm(int shmid)
{
    int n = shmctl(shmid, IPC_RMID, nullptr);
    assert(n != -1);
    (void)n;
}


#define SERVER 1
#define CLIENT 0
//封装
class Init
{
public:
    Init(int t)
        :type(t)
    {
        key_t key = getKey();
        if(type == SERVER)
            shmid = creatShm(key, SIZE);
        else
            shmid = getShm(key, SIZE);
        str = attachShm(shmid);
    }
    
    char* getStr()
    {
        return str;
    }

    ~Init()
    {
        detachShm(str);
        if(type == SERVER)
            delShm(shmid);
    }
private:
    char* str;
    int type;
    int shmid;
};
